package com.cloudkinto.service.role.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.cloudkinto.common.constant.SysConstant;
import com.cloudkinto.common.exception.BizException;
import com.cloudkinto.common.web.PageResult;
import com.cloudkinto.common.web.SingleResult;
import com.cloudkinto.entity.MenuDo;
import com.cloudkinto.dao.MenuDao;
import com.cloudkinto.service.role.MenuService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.cloudkinto.service.role.vo.menu.MenuAddReq;
import com.cloudkinto.service.role.vo.menu.MenuDetailRes;
import com.cloudkinto.service.role.vo.menu.MenuTreeRes;
import com.cloudkinto.service.role.vo.menu.MenuUpdateReq;
import org.springframework.stereotype.Service;
import org.springframework.beans.factory.annotation.Autowired;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
 * service实现类
 * </p>
 *
 * @author 张永远
 * @since 2020-05-29
 */
@Service
@Transactional(rollbackFor = Exception.class)//事务回滚
public class MenuServiceImpl extends ServiceImpl<MenuDao, MenuDo> implements MenuService {
    @Autowired
    private MenuDao dao;

    @Override
    public SingleResult addInit() {
        return SingleResult.success("");
    }

    @Override
    public MenuAddReq add(MenuAddReq req, Long userId) {
//        if(true){
//            throw new BizException("没有操作权限");
//        }
        addCheck(req);
        MenuDo entityDo = new MenuDo();
        BeanUtils.copyProperties(req, entityDo);
        if (req.getParentId() == null) {
            entityDo.setParentId(0L);
        }
        handlerAddBefore(entityDo, userId);
        dao.insert(entityDo);
        return req;
    }

    //新增校验
    private void addCheck(MenuAddReq req) {
        LambdaQueryWrapper<MenuDo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getName, req.getName()).eq(MenuDo::getType, req.getType()).ne(MenuDo::getUrl,"").eq(MenuDo::getSystem_,0);
        int count = this.baseMapper.selectCount(wrapper);
        if (count > 0) {
            throw new BizException("名称重复");
        }

        wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getUrl, req.getUrl()).eq(MenuDo::getType, req.getType()).eq(MenuDo::getSystem_,0);
        int urlCount = this.baseMapper.selectCount(wrapper);
        if (urlCount > 0) {
            throw new BizException("路径重复");
        }

        if(req.getType()==2 && StringUtils.isBlank(req.getUrl())){
            throw new BizException("请配置路由");
        }

        if (req.getParentId() != null) {
            MenuDo menuDo = this.baseMapper.selectById(req.getParentId());
            if (menuDo != null && menuDo.getType() == 2) {
                throw new BizException("父级目录为菜单，无法创建子目录或菜单");
            }
        }
    }

    //做新增之前要执行的业务定制处理
    private void handlerAddBefore(MenuDo entityDo, Long userId) {
        entityDo.setCreateBy(userId);
        entityDo.setCreateTime(new Date());
        entityDo.setDeleteFlag(SysConstant.DELETE_FLAG_FALSE);
    }

    @Override
    public SingleResult updateInit(Long id) {
        if(true){
            throw new BizException("没有操作权限");
        }
        MenuDetailRes res = new MenuDetailRes();
        MenuDo entityDo = dao.selectById(id);
        BeanUtils.copyProperties(entityDo, res);
        res.setParentId(entityDo.getParentId() == 0 ? null : entityDo.getParentId());
        MenuDo parentDo = dao.selectById(entityDo.getParentId());
        if (parentDo != null) {
            res.setParentName(parentDo.getName());
        }
        return SingleResult.success(res);
    }

    @Override
    public MenuUpdateReq update(MenuUpdateReq req, Long userId) {
        if(true){
            throw new BizException("没有操作权限");
        }
        updateCheck(req);
        MenuDo entityDo = dao.selectById(req.getId());
        BeanUtils.copyProperties(req, entityDo);
        handlerUpdateBefore(entityDo, userId);
        if (req.getParentId() == null) {
            entityDo.setParentId(0L);
        }
        dao.updateById(entityDo);
        return req;
    }

    //编辑校验
    private void updateCheck(MenuUpdateReq req) {
        LambdaQueryWrapper<MenuDo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getName, req.getName()).eq(MenuDo::getType, req.getType()).ne(MenuDo::getId, req.getId()).eq(MenuDo::getSystem_,0);
        int count = this.baseMapper.selectCount(wrapper);
        if (count > 0) {
            throw new BizException("名称重复");
        }

        wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getUrl, req.getUrl()).eq(MenuDo::getType, req.getType()).ne(MenuDo::getId, req.getId()).ne(MenuDo::getUrl,"").eq(MenuDo::getSystem_,0);
        int urlCount = this.baseMapper.selectCount(wrapper);
        if (urlCount > 0) {
            throw new BizException("路径重复");
        }

        if(req.getType()==2 && StringUtils.isBlank(req.getUrl())){
            throw new BizException("请配置路由");
        }

        if (req.getParentId() != null) {
            MenuDo menuDo = this.baseMapper.selectById(req.getParentId());
            if (menuDo != null && menuDo.getType() == 2) {
                throw new BizException("父级目录为菜单，无法创建子目录或菜单");
            }
        }
    }

    //做修改之前要执行的业务定制处理
    private void handlerUpdateBefore(MenuDo entityDo, Long userId) {
        entityDo.setUpdateBy(userId);
        entityDo.setUpdateTime(new Date());
    }


    @Override
    public int delete(Long id, Long userId) {
        if(true){
            throw new BizException("没有操作权限");
        }
        deleteCheck(id);
        MenuDo entityDo = dao.selectById(id);
        handlerDeleteBefore(entityDo, userId);
        dao.updateById(entityDo);
        return dao.deleteById(id);
    }

    private void deleteCheck(Long id) {
        LambdaQueryWrapper<MenuDo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getParentId, id);
        List<MenuDo> menuDoList = this.baseMapper.selectList(wrapper);
        if (!CollectionUtils.isEmpty(menuDoList)) {
            throw new BizException("该目录下面有菜单或目录，无法删除");
        }
    }


    //做删除之前要执行的业务定制处理
    private void handlerDeleteBefore(MenuDo entityDo, Long userId) {
        entityDo.setDeleteBy(userId);
        entityDo.setDeleteTime(new Date());
    }

    @Override
    public MenuDetailRes detail(Long id) {
        MenuDo entityDo = dao.selectById(id);
        MenuDetailRes res = new MenuDetailRes();
        BeanUtils.copyProperties(entityDo, res);
        res.setParentId(entityDo.getParentId() == 0 ? null : entityDo.getParentId());
        MenuDo parentDo = dao.selectById(entityDo.getParentId());
        if (parentDo != null) {
            res.setParentName(parentDo.getName());
        }

        return handlerDetailAfter(res);
    }

    //查询明细之后要执行的业务定制处理
    private MenuDetailRes handlerDetailAfter(MenuDetailRes res) {
        return res;
    }

    @Override
    public SingleResult pageInit() {
        return SingleResult.success("");
    }

    @Override
    public PageResult getListPage(Map<String, String> map) {
        QueryWrapper<MenuDo> wrapper = queryBuild(map);
        IPage<MenuDo> page = pageInit(map);
        page = this.page(page, wrapper);
        List<Object> resList = handlerListAfter(page.getRecords());
        PageResult result = new PageResult(resList, (int) page.getCurrent(), (int) page.getSize(), (int) page.getTotal(), (int) page.getPages());
        return result;
    }

    @Override
    public List<MenuTreeRes> menuTree() {
        List<MenuTreeRes> treeResList = new ArrayList<>();
        LambdaQueryWrapper<MenuDo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getParentId, 0L)
                .eq(MenuDo::getSystem_,0)
                .orderByAsc(MenuDo::getSort);
        List<MenuDo> menuDoList = dao.selectList(wrapper);
        if (!CollectionUtils.isEmpty(menuDoList)) {
            for (MenuDo menuDo : menuDoList) {
                MenuTreeRes treeRes = new MenuTreeRes();
                treeRes.setId(menuDo.getId());
                treeRes.setName(menuDo.getName());
                treeRes.setChildren(getChildren(treeRes.getId()));
                treeResList.add(treeRes);
            }
        }

        return treeResList;
    }

    private List<MenuTreeRes> getChildren(Long menuId) {
        List<MenuTreeRes> treeResList = new ArrayList<>();
        LambdaQueryWrapper<MenuDo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MenuDo::getParentId, menuId).
                eq(MenuDo::getSystem_,0).
                orderByAsc(MenuDo::getSort);
        List<MenuDo> menuDoList = dao.selectList(wrapper);
        if (!CollectionUtils.isEmpty(menuDoList)) {
            for (MenuDo menuDo : menuDoList) {
                MenuTreeRes res = new MenuTreeRes();
                res.setId(menuDo.getId());
                res.setName(menuDo.getName());
                res.setChildren(getChildren(res.getId()));
                treeResList.add(res);
            }
            return treeResList;
        }
        return null;
    }

    //查询分页列表之后对列表数据的处理
    private List<Object> handlerListAfter(List<MenuDo> list) {
        List<Object> result = new ArrayList<>();
        return result;
    }

    private IPage<MenuDo> pageInit(Map<String, String> map) {
        IPage<MenuDo> page = new Page();
        page.setSize(StringUtils.isNotBlank(map.get("pageSize")) ? Long.parseLong(map.get("pageSize")) : 10);
        page.setCurrent(StringUtils.isNotBlank(map.get("current")) ? Long.parseLong(map.get("current")) : 1);
        return page;
    }

    private QueryWrapper<MenuDo> queryBuild(Map<String, String> map) {
        QueryWrapper<MenuDo> wrapper = new QueryWrapper<>();
        if (StringUtils.isNotBlank(map.get("keyword"))) {
            wrapper.like("keyword", map.get("keyword"));
        }
        return wrapper;
    }


}
